import websocketClient, { MessageTypes } from '@/utils/websocket/websocket';
import { getToken } from '@/utils/auth'; // 假设你有获取token的方法

export default {
  install(Vue) {
    // 添加全局属性
    Vue.prototype.$websocket = websocketClient;
    Vue.prototype.$wsMessageTypes = MessageTypes;

    // 添加自动连接方法
    Vue.prototype.$connectWebsocket = function() {
      const token = getToken();
      if (token) {
        websocketClient.connect(token);
      }
    };
  }
};
